Sourced from optax's releases.

Optax 0.1.1

What's Changed

• Tweak the meta-learning example from the docs by @​copybara-service in deepmind/optax#233
• Fix small bugs in metalearning example. by @​copybara-service in deepmind/optax#236
• Do not reuse mini-batches between epochs in DPSGD example. by @​copybara-service in deepmind/optax#230
• Make the version of typing_extensions less constrained. by @​copybara-service in deepmind/optax#238
• [JAX] move example libraries from jax.experimental to jax.example_libraries by @​copybara-service in deepmind/optax#200
• Export ScaleByBeliefState by @​NeilGirdhar in deepmind/optax#239
• MultiStep optimizer: align parameter naming and type annotations of update function with signature of GradientTransform.update. by @​copybara-service in deepmind/optax#243
• Fix imports of datasets in examples folder. by @​copybara-service in deepmind/optax#242
• Enable example tests on github. Fix the bugs that were uncovered. by @​copybara-service in deepmind/optax#244
• Formatting. by @​copybara-service in deepmind/optax#249
• Add test for multi steps wrapper, verifying that the aggregated gradient is the mean of the input gradients. by @​copybara-service in deepmind/optax#255
• MultiStep optimizer wrapper: replace naive streaming average gradient implementation with numerically stabler one. by @​copybara-service in deepmind/optax#254
• Added ord, axis, and keepdims args to safe_norm by @​copybara-service in deepmind/optax#252
• Add badges and include RTD build into CI tests. by @​copybara-service in deepmind/optax#256
• Write a clearer doc-string for GradientTransformation by @​copybara-service in deepmind/optax#257
• Refactor clipping.py by @​copybara-service in deepmind/optax#260
• Implement split real norm by @​wdphy16 in deepmind/optax#241
• Monkey-patch sphinx to output correct type annotations for the most common cases (e.g. params, opt state) in the documentation. by @​copybara-service in deepmind/optax#266
• Improve docs by @​copybara-service in deepmind/optax#268
• Implement stateless wrapper. by @​n2cholas in deepmind/optax#246
• Replace _ with params to ensure you can always call init with named args. by @​copybara-service in deepmind/optax#270
• Improve docs. by @​copybara-service in deepmind/optax#269
• Add missing ` in two places. by @​copybara-service in deepmind/optax#273
• Add option to cache examples datasets after pre-processing. by @​copybara-service in deepmind/optax#272
• Fix an error in README.md rendering. by @​copybara-service in deepmind/optax#275
• Remove the old venv directory before testing the package. by @​copybara-service in deepmind/optax#289
• Fix Yogi optimizer by @​wdphy16 in deepmind/optax#288
• Bump ipython from 7.16.1 to 7.16.3 in /requirements by @​dependabot in deepmind/optax#286
• Clarifies optax.adamw(mask) parameter. by @​copybara-service in deepmind/optax#284
• Fix the link to the complex-valued optim proposal in RTD. by @​copybara-service in deepmind/optax#295
• Implement complex norm in optimizers by @​wdphy16 in deepmind/optax#279
• Change add_noise to match the target variance by scaling by its sqrt. by @​Rupt in deepmind/optax#294
• Minor tweaks to the optax documentation. by @​copybara-service in deepmind/optax#297
• Bump version to 0.1.1 from 0.1.0 by @​copybara-service in deepmind/optax#298

New Contributors

• @​wdphy16 made their first contribution in deepmind/optax#241
• @​dependabot made their first contribution in deepmind/optax#286
• @​Rupt made their first contribution in deepmind/optax#294

Full Changelog: google-deepmind/optax@v0.1.0...v0.1.1

• 4621c6a Bump version to 0.1.1 from 0.1.0
• 569a981 Minor tweaks to the optax documentation.
• 877ed66 Merge pull request #294 from Rupt:fix-add-noise-std
• 405bc0c Square eta for noise similar to pre-#294
• 7ddca8f rescale with a lambda default argument
• f41728d Address comments/nits
• 7a4ca0b Change add_noise to match the target variance by scaling by its sqrt.
• ee46275 Merge pull request #279 from wdphy16/complex_optimizer
• 1205713 Fix EMA bias correction
• b055efe Merge branch 'master' into complex_optimizer
• Additional commits viewable in compare view

You can trigger Dependabot actions by commenting on this PR:

• @dependabot rebase will rebase this PR
• @dependabot recreate will recreate this PR, overwriting any edits that have been made to it
• @dependabot merge will merge this PR after your CI passes on it
• @dependabot squash and merge will squash and merge this PR after your CI passes on it
• @dependabot cancel merge will cancel a previously requested merge and block automerging
• @dependabot reopen will reopen this PR if it is closed
• @dependabot close will close this PR and stop Dependabot recreating it. You can achieve the same result by closing it manually
• @dependabot ignore this major version will close this PR and stop Dependabot creating any more for this major version (unless you reopen the PR or upgrade to it yourself)
• @dependabot ignore this minor version will close this PR and stop Dependabot creating any more for this minor version (unless you reopen the PR or upgrade to it yourself)
• @dependabot ignore this dependency will close this PR and stop Dependabot creating any more for this dependency (unless you reopen the PR or upgrade to it yourself)